
const WS_ADDRESS = 'ws://192.168.231.20/socket/1'
export const useWebSocket = (handleMessage) => {
  const ws = new WebSocket(WS_ADDRESS)

  // ws.binaryType = 'arraybuffer'; //可将 WebSocket 对象的 binaryType 属性设为“blob”或“arraybuffer”。默认格式为“blob”（您不必在发送时校正 binaryType 参数）。
  const init = () => {
    ws.addEventListener('open', handleOpen, false)
    ws.addEventListener('close', handleClose, false)
    ws.addEventListener('error', handleError, false)
    // 消息处理方法，本处是以构造参数传入的方式更为灵活的处理消息
    ws.addEventListener('message', handleMessage, false)
  }
  // 连接建立时会调用的方法
  function handleOpen(e) {
    console.log('WebSocket open', e)
  }
  // 连接关闭时会调用的方法
  function handleClose(e) {
    console.log('WebSocket close', e)
  }
  // 连接出现异常时需要调用的方法
  function handleError(e) {
    console.log('WebSocket error', e)
  }

  init()
  return ws
}

export default { useWebSocket }
